package wells.leetcode.day20200714;

import java.util.HashMap;
import java.util.Map;

/**
 * <pre>
 * 给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。
 * 你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。 
 * 示例: 
 * 	给定 nums = [2, 7, 11, 15], target = 9
 * 	因为 nums[0] + nums[1] = 2 + 7 = 9 
 * 	所以返回 [0, 1]
 * 
 * </pre>
 * 
 * @author HuangSiwei
 * @date 2020/07/14
 */
public class Solution {
	public static void main(String[] args) {
		int[] nums = new int[] { 2, 7, 11, 15 };
		int target = 9;

		int[] twoSum = twoSum(nums, target);
		System.out.println(twoSum);
	}

	public static int[] twoSum(int[] nums, int target) {
		int[] result = new int[2];
		Map<Integer, Integer> tempMap = new HashMap<>();
		for (int i = 0; i < nums.length; i++) {
			int num = nums[i];
			Integer index = tempMap.get(target - num);
			if (index != null) {
				return new int[] { index, i };
			} else {
				tempMap.put(num, i);
			}
		}
		return result;
	}
}
